package com.wang.chatai.repository;

import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 会话历史id实现类
 */
@Slf4j
@Component
@RequiredArgsConstructor
public class InMemoryChatHistoryRepository implements ChatHistoryRepository {

    private final Map<String, List<String>> chatHistory = new HashMap<>();

    /**
     * 保存会话记录ID
     * @param type 业务类型，如：chat、service、pdf
     * @param chatId 会话ID
     */
    @Override
    public void save(String type, String chatId) {
        //如果存在type拿到对应的集合，不存在创建一个新的list
        List<String> chatIds = chatHistory.computeIfAbsent(type, k -> new ArrayList<>());
        //id已经存在了，直接返回
        if (chatIds.contains(chatId)) {
            return;
        }
        chatIds.add(chatId);
    }

    /**
     * 获取会话ID列表
     * @param type 业务类型，如：chat、service、pdf
     * @return 会话ID列表
     */
    @Override
    public List<String> getChatIds(String type) {
        return chatHistory.getOrDefault(type, List.of());
    }

}
